package top.buluoluo.productservice.dto.response;

import lombok.Data;
import top.buluoluo.productservice.entity.ProductCategory;

import java.util.ArrayList;
import java.util.List;

/**
 * 商品分类树形数据传输对象
 * 专门用于前端树形组件、层级导航等场景
 */
@Data
public class CategoryTreeDTO {
    private Long id;
    private Long parentId;
    private String name;
    private Integer level;
    private String iconUrl;
    private Integer status;
    private Long sortOrder;

    // 核心特性：支持嵌套子分类，形成树形结构
    private List<CategoryTreeDTO> children = new ArrayList<>();
    /**
     * 从实体类转换的工厂方法
     */
    public static CategoryTreeDTO convertToTreeDTO(ProductCategory category) {
        if (category == null) {
            return null;
        }
        CategoryTreeDTO dto = new CategoryTreeDTO();
        dto.setId(category.getId());
        dto.setParentId(category.getParentId());
        dto.setName(category.getName());
        dto.setLevel(category.getLevel());
        dto.setIconUrl(category.getIconUrl());
        dto.setStatus(category.getStatus());
        dto.setSortOrder(category.getSortOrder());
        return dto;
    }

    // 便捷方法：添加子分类
    public void addChild(CategoryTreeDTO child) {
        if (this.children == null) {
            this.children = new ArrayList<>();
        }
        this.children.add(child);
    }

    // 判断是否为根节点（一级分类）
    public boolean isRoot() {
        return parentId == null || parentId == 0;
    }

    // 判断是否有子分类
    public boolean hasChildren() {
        return children != null && !children.isEmpty();
    }

    // 获取子分类数量
    public int getChildrenCount() {
        return children != null ? children.size() : 0;
    }

    // 递归获取所有后代分类数量（包括子分类的子分类）
    public int getDescendantCount() {
        int count = 0;
        if (hasChildren()) {
            for (CategoryTreeDTO child : children) {
                count += 1 + child.getDescendantCount();
            }
        }
        return count;
    }
}